FROM pytorch/torchserve:latest

ADD infer.py /home/model-server/app/infer.py
ADD main.py  /home/model-server/app/main.py
ADD plot.py  /home/model-server/app/plot.py
ADD model_best.pth /home/model-server/app/model_best.pth

RUN pip install flask pandas numpy plotly tqdm pyarrow fastparquet -i https://pypi.tuna.tsinghua.edu.cn/simple

CMD ["python","/home/model-server/app/main.py"]
LABEL maintainer="timwcx@qq.com"
